tensor([[[[[0.0000, 0.0000],
           [0.0000, 0.0000]],

          [[0.0000, 0.0000],
           [0.0000, 0.0000]]],


         [[[0.0000, 0.0000],
           [0.0000, 0.0000]],

          [[1.0000, 1.0000],
           [1.0000, 1.0000]]]],



        [[[[0.0200, 0.0300],
           [0.0500, 0.0200]],

          [[0.4000, 0.4100],
           [0.4100, 0.4100]]],


         [[[0.4000, 0.4000],
           [0.4000, 0.4100]],

          [[0.0000, 0.0000],
           [0.0000, 0.0000]]]],



        [[[[0.9700, 0.9500],
           [0.9400, 0.9700]],

          [[0.0100, 0.0000],
           [0.0000, 0.0100]]],


         [[[0.0100, 0.0000],
           [0.0100, 0.0100]],

          [[0.0000, 0.0000],
           [0.0000, 0.0000]]]],



        [[[[0.0100, 0.0200],
           [0.0200, 0.0100]],

          [[0.5900, 0.5900],
           [0.5900, 0.5800]]],


         [[[0.6000, 0.6000],
           [0.5900, 0.5800]],

          [[0.0000, 0.0000],
           [0.0000, 0.0000]]]]], grad_fn=<RoundBackward1>)